/*
 * Copyright (C) 2017 Ortega Froysa, Nicolás <nortega@themusicinnoise.net>
 * Author: Ortega Froysa, Nicolás <nortega@themusicinnoise.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "neocomm.h"

#include "globals.hpp"
#include "error.hpp"

std::map<std::string, struct channel_info> channels;

static std::mt19937_64 rand_dev { dht::crypto::random_device{}() };
static std::uniform_int_distribution<dht::Value::Id> rand_id;

static std::map<time_t, int> sent_messages;

struct NeoComm_message *NeoComm_get_next_message(const char *channel_name) {
	const std::string s_chan_name(channel_name);
	if(channels[s_chan_name].msgs.empty())
		return NULL;
	const dht::ImMessage *new_msg = &channels[s_chan_name].msgs.front();
	struct NeoComm_message *msg =
		static_cast<struct NeoComm_message*>(
				malloc(sizeof(struct NeoComm_message)));

	msg->msg = static_cast<char*>(malloc(new_msg->msg.size()));
	strcpy(msg->msg, new_msg->msg.c_str());
	msg->sent = new_msg->date;
	msg->from.nick = NULL; // TODO: Implement this!
	msg->from.hash = static_cast<char*>(
			malloc(strlen(new_msg->from.to_c_str())));
	strcpy(msg->from.hash, new_msg->from.to_c_str());
	channels[s_chan_name].msgs.pop();

	return msg;
}

void NeoComm_free_message(struct NeoComm_message *msg) {
	free(msg->msg);
	free(msg->from.hash);
	free(msg);
}

time_t NeoComm_send_message(const char *channel_name, const char *message) {
	if(not node.isRunning())
	{
		add_error("NeoComm must be initialized.");
		return 0;
	}
	const dht::InfoHash chan_hash = channels[channel_name].hash;
	const time_t now =
		std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());

	if(channels.find(channel_name) == channels.end())
	{
		add_error("Not connected to channel");
		return 0;
	}
	sent_messages[now] = NC_PENDING;
	node.putSigned(chan_hash,
			dht::ImMessage(rand_id(rand_dev), message, now), [&](bool sent) {
				sent_messages[now] = sent ? NC_GOOD : NC_BAD;
			});
	return now;
}

// TODO: Devise a way so that the message can be resent.
int NeoComm_check_message(const time_t token) {
	if(sent_messages.find(token) == sent_messages.end())
	{
		add_error("No message found with that token.");
		return 0;
	}
	const int status = sent_messages[token];
	sent_messages.erase(token);
	return status;
}
